package com.allm.ai.common.repository;

import java.util.List;

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import com.allm.ai.common.entity.AiModelTag;

/**
 * AI模型标签关联数据访问层
 */
@Repository
public interface AiModelTagRepository extends JpaRepository<AiModelTag, Long> {

    /**
     * 根据模型ID查找所有启用的关联
     */
    List<AiModelTag> findByModelIdAndIsActiveTrue(Long modelId);

    /**
     * 根据模型ID查找所有关联（包括禁用的）
     */
    List<AiModelTag> findByModelId(Long modelId);

    /**
     * 根据标签ID查找所有启用的关联
     */
    List<AiModelTag> findByTagIdAndIsActiveTrue(Long tagId);

    /**
     * 根据标签ID查找所有关联（包括禁用的）
     */
    List<AiModelTag> findByTagId(Long tagId);

    /**
     * 软删除模型的所有标签关联
     */
    @Modifying
    @Query("UPDATE AiModelTag mt SET mt.isActive = false WHERE mt.modelId = :modelId")
    void softDeleteByModelId(@Param("modelId") Long modelId);

    /**
     * 硬删除模型的所有标签关联
     */
    @Modifying
    @Query("DELETE FROM AiModelTag mt WHERE mt.modelId = :modelId")
    void deleteByModelId(Long modelId);

    /**
     * 软删除指定的标签关联
     */
    @Modifying
    @Query("UPDATE AiModelTag mt SET mt.isActive = false WHERE mt.modelId = :modelId AND mt.tagId IN :tagIds")
    void softDeleteByModelIdAndTagIdIn(@Param("modelId") Long modelId, @Param("tagIds") List<Long> tagIds);

    /**
     * 硬删除指定的标签关联
     */
    @Modifying
    @Query("DELETE FROM AiModelTag mt WHERE mt.modelId = :modelId AND mt.tagId IN :tagIds")
    void deleteByModelIdAndTagIdIn(Long modelId, List<Long> tagIds);

    /**
     * 检查模型和标签的关联是否存在且启用
     */
    boolean existsByModelIdAndTagIdAndIsActiveTrue(Long modelId, Long tagId);

    /**
     * 检查模型和标签的关联是否存在（包括禁用的）
     */
    boolean existsByModelIdAndTagId(Long modelId, Long tagId);

    /**
     * 根据模型ID获取启用的标签ID列表
     */
    @Query("SELECT mt.tagId FROM AiModelTag mt WHERE mt.modelId = :modelId AND mt.isActive = true")
    List<Long> findActiveTagIdsByModelId(@Param("modelId") Long modelId);

    /**
     * 根据标签ID获取启用的模型ID列表
     */
    @Query("SELECT mt.modelId FROM AiModelTag mt WHERE mt.tagId = :tagId AND mt.isActive = true")
    List<Long> findActiveModelIdsByTagId(@Param("tagId") Long tagId);

    /**
     * 根据模型ID和标签ID查找关联
     */
    List<AiModelTag> findByModelIdAndTagId(Long modelId, Long tagId);
}